package com.example.financeapi.utils;

import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.redis.connection.RedisConnection;
import org.springframework.data.redis.connection.RedisConnectionFactory;
import org.springframework.data.redis.core.*;
import org.springframework.stereotype.Component;

import java.util.*;
import java.util.concurrent.TimeUnit;

/**
 * Redis 工具类
 * @author Alvin
 */
@Slf4j
@Component
public class RedisUtils {

	@Autowired
	private StringRedisTemplate redisTemplate;

	//@Autowired
	//private RedisTemplate<String, Object> redisTemplateTwo;

	/**
	 * 实现命令：TTL key，以秒为单位，返回给定 key的剩余生存时间(TTL, time to live)。
	 *
	 * @param key
	 * @return
	 */
	public long ttl(String key) {
		return redisTemplate.getExpire(key);
	}

	/**
	 * 根据 key 获取过期时间
	 *
	 * @param key 键 不能为null
	 * @return 时间(秒) 返回0代表为永久有效
	 */
	public long getTTL(Object key) {
		return redisTemplate.getExpire((String) key, TimeUnit.SECONDS);
	}

	/**
	 * 实现命令：expire 设置过期时间，单位秒
	 *
	 * @param key
	 * @return
	 */
	public void expire(String key, long timeout) {
		redisTemplate.expire(key, timeout, TimeUnit.SECONDS);
	}

	/**
	 * 实现命令：INCR key，增加key一次
	 *
	 * @param key
	 * @return
	 */
	public long incr(String key, long delta) {
		return redisTemplate.opsForValue().increment(key, delta);
	}

	/**
	 * 实现命令：KEYS pattern，查找所有符合给定模式 pattern的 key
	 * @return
	 */
	public Set<String> keys(String pattern) {
		return redisTemplate.keys(pattern);
	}

	/**
	 * 删除对应的value
	 * @param key
	 */
	public boolean remove(final String key) {
		if (exists(key)) {
			return redisTemplate.delete(key);
		}
		return false;
	}

	/**
	 * 实现命令：DEL key，删除一个key
	 *
	 * @param key
	 */
	public void del(String key) {
		redisTemplate.delete(key);
	}

	/**
	 * 删除缓存
	 * @param keys 可以传一个值 或多个
	 */
	public void del(String... keys) {
		if (keys != null && keys.length > 0) {
			if (keys.length == 1) {
				boolean result = redisTemplate.delete(keys[0]);
				log.debug("--------------------------------------------");
				log.debug(new StringBuilder("删除缓存：").append(keys[0]).append("，结果：").append(result).toString());
				log.debug("--------------------------------------------");
			} else {
				List keySet = new ArrayList<>();
				for (String key : keys) {
					keySet.add(redisTemplate.keys(key));
				}
				Long count =redisTemplate.delete(keySet);
				log.debug("--------------------------------------------");
				log.debug("成功删除缓存：" + keySet.toString());
				log.debug("缓存删除数量：" + count + "个");
				log.debug("--------------------------------------------");
			}
		}
	}

	/**
	 * @param prefix 前缀
	 * @param ids    id
	 */
	public void delByKeys(String prefix, Set<Long> ids) {
		List keys = new ArrayList();
		for (Long id : ids) {
			keys.addAll(redisTemplate.keys(new StringBuffer(prefix).append(id).toString()));
		}
		long count = redisTemplate.delete(keys);
		// 此处提示可自行删除
		log.debug("--------------------------------------------");
		log.debug("成功删除缓存：" + keys.toString());
		log.debug("缓存删除数量：" + count + "个");
		log.debug("--------------------------------------------");
	}

	/**
	 * 实现命令：SET key value，设置一个key-value（将字符串值 value关联到 key）
	 *
	 * @param key
	 * @param value
	 */
	public void set(String key, String value) {
		redisTemplate.opsForValue().set(key, value);
	}

	/**
	 * 实现命令：SET key value EX seconds，设置key-value和超时时间（秒）
	 *
	 * @param key
	 * @param value
	 * @param timeout
	 *            （以秒为单位）
	 */
	public void set(String key, String value, long timeout) {
		redisTemplate.opsForValue().set(key, value, timeout, TimeUnit.SECONDS);
	}

	/**
	 * 实现命令：GET key，返回 key所关联的字符串值。
	 *
	 * @param key
	 * @return value
	 */
	public String get(String key) {
		return redisTemplate.opsForValue().get(key);
	}

	/**
	 * 实现命令：HSET key field value，将哈希表 key中的域 field的值设为 value
	 *
	 * @param key
	 * @param field
	 * @param value
	 */
	public void hset(String key, String field, Object value) {
		redisTemplate.opsForHash().put(key, field, value);
	}

	/**
	 * 实现命令：HGET key field，返回哈希表 key中给定域 field的值
	 *
	 * @param key
	 * @param field
	 * @return
	 */
	public String hget(String key, String field) {
		return (String) redisTemplate.opsForHash().get(key, field);
	}

	/**
	 * 实现命令：HDEL key field [field ...]，删除哈希表 key 中的一个或多个指定域，不存在的域将被忽略。
	 * 删除hash表中的值
	 * @param key  键 不能为null
	 * @param fields 可以使多个 不能为null
	 */
	public void hdel(String key, Object... fields) {
		redisTemplate.opsForHash().delete(key, fields);
	}

	/**
	 * 实现命令：HGETALL key，返回哈希表 key中，所有的域和值。
	 *
	 * @param key
	 * @return
	 */
	public Map<Object, Object> hgetall(String key) {
		return redisTemplate.opsForHash().entries(key);
	}

	/**
	 * 实现命令：LPUSH key value，将一个值 value插入到列表 key的表头
	 *
	 * @param key
	 * @param value
	 * @return 执行 LPUSH命令后，列表的长度。
	 */
	public long lpush(String key, String value) {
		return redisTemplate.opsForList().leftPush(key, value);
	}

	/**
	 * 实现命令：LPOP key，移除并返回列表 key的头元素。
	 * @param key
	 * @return 列表key的头元素。
	 */
	public String lpop(String key) {
		return redisTemplate.opsForList().leftPop(key);
	}

	/**
	 * 实现命令：RPUSH key value，将一个值 value插入到列表 key的表尾(最右边)。
	 * @param key
	 * @param value
	 * @return 执行 LPUSH命令后，列表的长度。
	 */
	public long rpush(String key, String value) {
		return redisTemplate.opsForList().rightPush(key, value);
	}

	/**
	 * 判断缓存中是否有对应的value
	 * @param key
	 * @return
	 */
	public boolean exists(final String key) {
		return redisTemplate.hasKey(key);
	}

	/**
	 * 查找匹配key
	 * @param pattern key
	 * @return /
	 */
	public List<String> scan(String pattern) {
		ScanOptions options = ScanOptions.scanOptions().match(pattern).build();
		RedisConnectionFactory factory = redisTemplate.getConnectionFactory();
		RedisConnection rc = Objects.requireNonNull(factory).getConnection();
		Cursor<byte[]> cursor = rc.scan(options);
		List<String> result = new ArrayList<>();
		while (cursor.hasNext()) {
			result.add(new String(cursor.next()));
		}
		try {
			RedisConnectionUtils.releaseConnection(rc, factory);
		} catch (Exception e) {
			e.printStackTrace();
		}
		return result;
	}

	/**
	 * 分页查询 key
	 * @param patternKey key
	 * @param page       页码
	 * @param size       每页数目
	 * @return /
	 */
	public List<String> findKeysForPage(String patternKey, int page, int size) {
		ScanOptions options = ScanOptions.scanOptions().match(patternKey).build();
		RedisConnectionFactory factory = redisTemplate.getConnectionFactory();
		RedisConnection rc = Objects.requireNonNull(factory).getConnection();
		Cursor<byte[]> cursor = rc.scan(options);
		List<String> result = new ArrayList<>(size);
		int tmpIndex = 0;
		int fromIndex = page * size;
		int toIndex = page * size + size;
		while (cursor.hasNext()) {
			if (tmpIndex >= fromIndex && tmpIndex < toIndex) {
				result.add(new String(cursor.next()));
				tmpIndex++;
				continue;
			}
			// 获取到满足条件的数据后,就可以退出了
			if (tmpIndex >= toIndex) {
				break;
			}
			tmpIndex++;
			cursor.next();
		}
		try {
			RedisConnectionUtils.releaseConnection(rc, factory);
		} catch (Exception e) {
			e.printStackTrace();
		}
		return result;
	}

	//===============================Map=================================
	/**
	 * 存Hash类型数据
	 * @param key
	 * @param value
	 * @param time
	 * @return
	 */
	public boolean setMap(String key, Map<String,Object> value, long time) {
		try {
			redisTemplate.opsForHash().putAll(key,value);
			redisTemplate.expire(key, time, TimeUnit.SECONDS);
			return true;
		} catch (Exception e) {
			e.printStackTrace();
			return false;
		}
	}

	/**
	 * 取Hash类型数据
	 * @param key
	 * @return
	 */
	public Object getMap(String key) {
		return key == null ? null : redisTemplate.opsForHash().entries(key);
	}

	//===============================list=================================
	/**
	 * 获取list缓存的内容
	 * @param key 键
	 * @param start 开始
	 * @param end 结束  0 到 -1代表所有值
	 * @return
	 */
	public List<String> lGet(String key, long start, long end){
		try {
			return redisTemplate.opsForList().range(key, start, end);
		} catch (Exception e) {
			e.printStackTrace();
			return null;
		}
	}

	/**
	 * 获取list缓存的长度
	 * @param key 键
	 * @return
	 */
	public long lGetListSize(String key){
		try {
			return redisTemplate.opsForList().size(key);
		} catch (Exception e) {
			e.printStackTrace();
			return 0;
		}
	}

	/**
	 * 通过索引 获取list中的值
	 * @param key 键
	 * @param index 索引  index>=0时， 0 表头，1 第二个元素，依次类推；index<0时，-1，表尾，-2倒数第二个元素，依次类推
	 * @return
	 */
	public Object lGetIndex(String key,long index){
		try {
			return redisTemplate.opsForList().index(key, index);
		} catch (Exception e) {
			e.printStackTrace();
			return null;
		}
	}

	/**
	 * 将list放入缓存
	 * @param key 键
	 * @param value 值
	 * @return
	 */
	public boolean lSet(String key, Object value) {
		try {
			redisTemplate.opsForList().rightPush(key, (String) value);
			return true;
		} catch (Exception e) {
			e.printStackTrace();
			return false;
		}
	}

	/**
	 * 将list放入缓存
	 * @param key 键
	 * @param value 值
	 * @param time 时间(秒)
	 * @return
	 */
	public boolean lSet(String key, Object value, long time) {
		try {
			redisTemplate.opsForList().rightPush(key, (String) value);
			if (time > 0) {
				expire(key, time);
			}
			return true;
		} catch (Exception e) {
			e.printStackTrace();
			return false;
		}
	}

	/**
	 * 将list放入缓存
	 * @param key 键
	 * @param value 值
	 * @return
	 */
	public boolean lSet(String key, List<Object> value) {
		try {
			redisTemplate.opsForList().rightPushAll(key, String.valueOf(value));
			return true;
		} catch (Exception e) {
			e.printStackTrace();
			return false;
		}
	}

	/**
	 * 将list放入缓存
	 * @param key 键
	 * @param value 值
	 * @param time 时间(秒)
	 * @return
	 */
	public boolean lSet(String key, List<Object> value, long time) {
		try {
			redisTemplate.opsForList().rightPushAll(key, String.valueOf(value));
			if (time > 0) {
				expire(key, time);
			}
			return true;
		} catch (Exception e) {
			e.printStackTrace();
			return false;
		}
	}

	/**
	 * 根据索引修改list中的某条数据
	 * @param key 键
	 * @param index 索引
	 * @param value 值
	 * @return
	 */
	public boolean lUpdateIndex(String key, long index,Object value) {
		try {
			redisTemplate.opsForList().set(key, index, (String) value);
			return true;
		} catch (Exception e) {
			e.printStackTrace();
			return false;
		}
	}

	/**
	 * 移除N个值为value
	 * @param key 键
	 * @param count 移除多少个
	 * @param value 值
	 * @return 移除的个数
	 */
	public long lRemove(String key,long count,Object value) {
		try {
			Long remove = redisTemplate.opsForList().remove(key, count, value);
			return remove;
		} catch (Exception e) {
			e.printStackTrace();
			return 0;
		}
	}

	/**
	 * 使用Redis的消息队列
	 * @param channel
	 * @param message 消息内容
	 */
	public void convertAndSend(String channel, Object message){
		redisTemplate.convertAndSend(channel,message);
	}

	/**
	 *将数据添加到Redis的list中（从右边添加）
	 * @param listKey
	 * @param time 有效期
	 * @param values 待添加的数据
	 */
	public void addToListRight(String listKey, long time, TimeUnit timeUnit, String... values) {
		//绑定操作
		BoundListOperations<String, String> boundValueOperations = redisTemplate.boundListOps(listKey);
		//插入数据
		boundValueOperations.rightPushAll(values);
		//设置过期时间
		boundValueOperations.expire(time,timeUnit);
	}
	/**
	 * 根据起始结束序号遍历Redis中的list
	 * @param listKey
	 * @param start  起始序号
	 * @param end  结束序号
	 * @return
	 */
	public List<String> rangeList(String listKey, long start, long end) {
		//绑定操作
		BoundListOperations<String, String> boundValueOperations = redisTemplate.boundListOps(listKey);
		//查询数据
		return boundValueOperations.range(start, end);
	}
	/**
	 * 弹出右边的值 --- 并且移除这个值
	 * @param listKey
	 */
	public Object rifhtPop(String listKey){
		//绑定操作
		BoundListOperations<String, String> boundValueOperations = redisTemplate.boundListOps(listKey);
		return boundValueOperations.rightPop();
	}

	//===============================hyperloglog=================================
	/**
	 * 在当数据量达到百万时，最佳的存储方式是 hyperloglog 用户日活月活统计
	 * hyperloglog 算法:
	 *   添加一条记录
	 * @param key
	 * @param obj
	 * @return
	 */
	public boolean hyperLogAdd(String key, String obj) {
		return redisTemplate.opsForHyperLogLog().add(key, obj) > 0;
	}

	/**
	 * hyperloglog 算法:
	 *   pfcount 非精准统计 key的计数
	 * @param key
	 * @return
	 */
	public long hyperLogCount(String key) {
		return redisTemplate.opsForHyperLogLog().size(key);
	}

	/**
	 * hyperloglog 算法:
	 *   将多个 hyperloglog 合并成一个新的 hyperloglog
	 * @param out
	 * @param key
	 * @return
	 */
	public boolean hyperLogMerge(String out, String... key) {
		// pfmerge out key1 key2  ---> 将key1 key2 合并成一个新的hyperloglog out
		return redisTemplate.opsForHyperLogLog().union(out, key) > 0;
	}

}